import requests

def get_embedding(text: str) -> list:
    """使用 Ollama 的 HTTP API 获取文本嵌入"""
    response = requests.post(
        'http://localhost:11434/api/embeddings',
        json={
            'model': 'bge-m3:latest',
            'prompt': text
        }
    )
    # return response.json()['embedding']
    return response.json()

# 示例使用
text = "这是一个示例文本"
embedding = get_embedding(text)
print(embedding)
# [-1.4269912242889404, 0.9092104434967041, ...]
